package com.ruoyi.job.util;

import java.text.ParseException;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.format.DateTimeFormatter;
import java.util.Date;
import org.quartz.CronExpression;

/**
 * cron表达式工具类
 * 
 * @author ruoyi
 *
 */
public class CronUtils
{
    /**
     * 返回一个布尔值代表一个给定的Cron表达式的有效性
     *
     * @param cronExpression Cron表达式
     * @return boolean 表达式是否有效
     */
    public static boolean isValid(String cronExpression)
    {
        return CronExpression.isValidExpression(cronExpression);
    }

    /**
     * 返回一个字符串值,表示该消息无效Cron表达式给出有效性
     *
     * @param cronExpression Cron表达式
     * @return String 无效时返回表达式错误描述,如果有效返回null
     */
    public static String getInvalidMessage(String cronExpression)
    {
        try
        {
            new CronExpression(cronExpression);
            return null;
        }
        catch (ParseException pe)
        {
            return pe.getMessage();
        }
    }

    /**
     * 返回下一个执行时间根据给定的Cron表达式
     *
     * @param cronExpression Cron表达式
     * @return Date 下次Cron表达式执行时间
     */
    public static Date getNextExecution(String cronExpression)
    {
        try
        {
            CronExpression cron = new CronExpression(cronExpression);
            return cron.getNextValidTimeAfter(new Date(System.currentTimeMillis()));
        }
        catch (ParseException e)
        {
            throw new IllegalArgumentException(e.getMessage());
        }
    }

    public static void main(String[] args) {
        //仅一次   05 06 11 31 03 ? 2022
        String onlyOnce = onlyOnce("2022-03-31 11:06:05");
        System.out.println(onlyOnce);
        //每天    05 51 09 * * ?
        String everyDay = everyDay("09:51:05");
        System.out.println(everyDay);
        //每周    05 51 09 ? * 1
        String everyWeek = everyWeek("09:51:05","1");
        System.out.println(everyWeek);
        //每月    05 51 09 31 * ?
        String everyMonth = everyMonth("2022-03-31 09:51:05");
        System.out.println(everyMonth);
        //每年    05 51 09 31 03 *
        String everyYear = everyYear("2022-03-31 09:51:05");
        System.out.println(everyYear);
    }


    /**
     * 仅一次
     *
     * @param dateStr   2022-03-31 09:51:05
     * @return
     */
    public static String onlyOnce(String dateStr) {
        LocalDateTime time = LocalDateTime.parse(dateStr, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
        String cronStr = time.format(DateTimeFormatter.ofPattern("ss mm HH dd MM ? yyyy"));
        return cronStr;
    }

    /**
     * 每天
     * @param timeStr   09:51:05
     * @return
     */
    public static String everyDay(String timeStr) {
        LocalTime time = LocalTime.parse(timeStr, DateTimeFormatter.ofPattern("HH:mm:ss"));
        String cronStr = time.format(DateTimeFormatter.ofPattern("ss mm HH * * ?"));
        return cronStr;
    }

    /**
     * 每周
     * @param timeStr   09:51:05
     * @param week      1代表周日 7代表周六 可选值：1、2、3、4、5、6、7、1-5、1-7
     * @return
     */
    public static String everyWeek(String timeStr, String week) {
        LocalTime time = LocalTime.parse(timeStr, DateTimeFormatter.ofPattern("HH:mm:ss"));
        String cronStr = time.format(DateTimeFormatter.ofPattern("ss mm HH ? * " + week));
        return cronStr;
    }

    /**
     * 每月
     * @param dateStr   2022-03-31 09:51:05
     * @return
     */
    public static String everyMonth(String dateStr) {
        LocalDateTime time = LocalDateTime.parse(dateStr, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
        String cronStr = time.format(DateTimeFormatter.ofPattern("ss mm HH dd * ?"));
        return cronStr;
    }

    /**
     * 每年
     * @param dateStr   2022-03-31 09:51:05
     * @return
     */
    public static String everyYear(String dateStr) {
        LocalDateTime time = LocalDateTime.parse(dateStr, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
        String cronStr = time.format(DateTimeFormatter.ofPattern("ss mm HH dd MM *"));
        return cronStr;
    }
}
